from page_object_demo.Base.base_page import BasePage
class BaiduPage(BasePage):
    url = "https://www.baidu.com"

    def search_input(self, search_key):
        self.by_id("kw").send_keys(search_key)

    def search_button(self):
        self.by_id("su").click()